"""
存放文件的与数据库打交道
"""

import os
import pickle
from conf import settings


# 查看当前用户数据
def select_data(cls, username):
    # 获取当前类的名称
    class_name = cls.__name__
    # 判断当前类的文件夹是否存在
    class_path = os.path.join(
        settings.DB_DIR_path, class_name
    )

    # 不存在则创建
    if not os.path.exists(class_path):
        os.mkdir(class_path)

    # 拼接当前用户文件的路径
    user_path = os.path.join(
        class_path, username
    )

    if os.path.exists(user_path):
        with open(user_path, 'rb') as f:
            res = pickle.load(f)
            return res


# 保存数据
def save_data(obj):
    # 获取当前类的名称
    class_name = obj.__class__.__name__
    # 判断当前类的文件夹是否存在
    class_path = os.path.join(
        settings.DB_DIR_path, class_name
    )

    # 不存在则创建
    if not os.path.exists(class_path):
        os.mkdir(class_path)

    # 拼接当前用户文件的路径
    user_path = os.path.join(
        class_path, obj.user
    )


    with open(user_path, 'wb') as f:
        pickle.dump(obj, f)